import torch as th
from tensorboardX import SummaryWriter
from torchsummary import summary

actor = th.load('./checkpoint_reward_0_1_0/agent_actor_100.pkl')
critic = th.load('./checkpoint_reward_0_1_0/agent_critic_100.pkl')
        

print(actor)
 
'''       
with SummaryWriter(comment='actor') as w:
    w.add_graph(actor) 
with SummaryWriter(comment='critic') as w:
    w.add_graph(critic) 
'''